/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
/*
 * Copyright (C) 2000, O'Reilly & Associates, Inc.
 * The code in this file may be copied, distributed, and reused,
 * provided that you do not remove this copyright notice.
 * O'Reilly & Associates assumes no responsibility for damages
 * resulting from the use of this code.
 */

package cgl.narada.samples.jms;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Properties;

import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.Topic;
import javax.jms.TopicConnection;
import javax.jms.TopicConnectionFactory;
import javax.jms.TopicPublisher;
import javax.jms.TopicSession;
import javax.jms.TopicSubscriber;

import cgl.narada.jms.NBJmsInitializer;
import cgl.narada.util.PropertiesHelp;
import cgl.narada.wsinfra.schemas.eventing.SubscribeDocument.Subscribe;

/**
 * This program is a demonstration of the features support by the new JMS
 * implementation and is adapted from the Chat example contained in O'Reilly's
 * Java Message Service Book.
 * 
 * @author Shrideep Pallickara $Revision$ $Date$
 */

public class Chat implements javax.jms.MessageListener {
    private TopicSession pubSession;

    private TopicSession subSession;

    private TopicPublisher publisher;

    private TopicConnection connection;

    private String topicString = "chatTopic=/streams/chat/private/Text";
    
    private String userName;

    public Chat(Properties connectionProperties, String transportType, String userName)
                                                                                       throws Exception {
        NBJmsInitializer ini;

        ini = new NBJmsInitializer(connectionProperties, transportType, userName
                .hashCode());
        // Lookup a JMS connection factory
        TopicConnectionFactory conFactory = (TopicConnectionFactory) ini.lookup();
        initializeChatSession(conFactory, userName);
    }

    public void initializeChatSession(TopicConnectionFactory conFactory, String username)
                                                                                         throws Exception {
        // Create a JMS connection
        TopicConnection connection = conFactory
                .createTopicConnection("guest", "password");

        // Create a JMS session object
        TopicSession pubSession = connection.createTopicSession(false,
                                                                Session.AUTO_ACKNOWLEDGE);
        TopicSession subSession = connection.createTopicSession(false,
                                                                Session.AUTO_ACKNOWLEDGE);

        
        // Lookup a JMS topic
        Topic chatTopic = pubSession.createTopic(topicString);

        System.out.println("Will be publishing to topic [" + topicString + "]");
        // Create a JMS publisher and subscriber
        TopicPublisher publisher = pubSession.createPublisher(chatTopic);
        TopicSubscriber subscriber = subSession.createSubscriber(chatTopic);
        //TopicSubscriber subscriber2 = subSession.createSubscriber(chatTopic);
       
        
        // Set a JMS message listener
        subscriber.setMessageListener(this);
        //subscriber2.setMessageListener(this);
        // Intialize the Chat application
        set(connection, pubSession, subSession, publisher, username);

        // Start the JMS connection; allows messages to be delivered
        connection.start();
    }

    public void startConnection() throws JMSException {
        connection.start();
    }

    public void stopConnection() throws JMSException {
        connection.stop();
    }

    /* Initializes the instance variables */
    public void set(TopicConnection con, TopicSession pubSess, TopicSession subSess,
                    TopicPublisher pub, String username) {
        this.connection = con;
        this.pubSession = pubSess;
        this.subSession = subSess;
        this.publisher = pub;
        this.userName = username;
    }

    /* Receive message from topic subscriber */
    public void onMessage(Message message) {
        try {
            TextMessage textMessage = (TextMessage) message;
            String text = textMessage.getText();
            System.out.println(text);
        } catch (JMSException jmse) {
            jmse.printStackTrace();
        }
    }

    /* Create and send message using topic publisher */
    public void writeMessage(String text) throws JMSException {
        TextMessage message = pubSession.createTextMessage();
        message.setText("[" + userName + "] : " + text);
        publisher.publish(message);
    }

    public void unsubscribe() {
        System.out.println("Unsubscribing ...");
        try {
            subSession.unsubscribe(topicString);
        } catch (JMSException jmsEx) {
            System.out.println("Problems unsubscribing!" + jmsEx);
            jmsEx.printStackTrace();
        }
    }
    
    
    public void subscribe() {
        System.out.println("Unsubscribing ...");
        try {
            Topic chatTopic = pubSession.createTopic(topicString);
            TopicSubscriber subscriber = subSession.createSubscriber(chatTopic);
            subscriber.setMessageListener(this);
        } catch (JMSException jmsEx) {
            System.out.println("Problems unsubscribing!" + jmsEx);
            jmsEx.printStackTrace();
        }
        
    }
    
    /* Close the JMS connection */
    public void close() throws JMSException {
        connection.close();
    }

    /* Run the Chat client */
    public static void main(String[] args) {
        String transport = "niotcp";

        try {
            if (args.length > 4 || args.length < 3) {
                System.out.println("Usage: java cgl.narada.samples.jms.Chat <hostname> "
                                   + "<portnum> <userName> [niotcp|tcp|http]");
                System.exit(0);
            } else if (args.length == 4) {
                transport = args[3];
            }

            System.out.println("Type h for usage options");
            Properties props = new Properties();
            /** These properties pertain to setting up a TCP link */
            props.put("hostname", args[0]);
            props.put("portnum", args[1]);

            /**
             * Depending on the transport you choose, you can vary the transport
             * protocol being used. Here we use "tcp". You could also use "udp",
             * "multicast" or "ssl"
             */

            if (transport.equals("http") || transport.equals("ssl")) {
                // String configFileName = "config/BrokerConfiguration.txt";
                String configFileName = "transportTest.properties";
                PropertiesHelp help = new PropertiesHelp(configFileName, transport);
                help.addProperties(props);
            }

            Chat chat = new Chat(props, transport, args[2]);

            // read from command line
            BufferedReader commandLine = new BufferedReader(
                                                            new InputStreamReader(
                                                                                  System.in));

            // loop until the word "exit" is typed
            int tracker = 0;
            while (true) {
                String s = commandLine.readLine();
                if (s == null) {
                    chat.close();
                    break;
                }
                if (s.equalsIgnoreCase("h")) {
                    System.out.println("Simply type messages in for normal chat");
                    System.out.println("stop    - Stop the connection. Messages cached");
                    System.out.println("start   - Starts connection. Cache released");
                    System.out.println("exit    - Exits the application");
                    continue;
                }
                if (s.equalsIgnoreCase("start")) {
                    chat.startConnection();
                    continue;
                }
                if (s.equalsIgnoreCase("stop")) {
                    chat.stopConnection();
                    continue;
                }
                
                if (s.equalsIgnoreCase("unsub")) {
                    chat.unsubscribe();
                    continue;
                }
                
                if (s.equalsIgnoreCase("sub")) {
                    chat.subscribe();
                    continue;
                }
                
                if (s.equalsIgnoreCase("exit")) {
                    chat.close(); // close down connection
                    break;
                }
                tracker++;
                chat.writeMessage(s + tracker);
            }
            /** end while-true */
            System.out.println("Chat: exiting ");
        } catch (Exception e) {
            e.printStackTrace();
        }
        System.out.println("Chat: " + Thread.currentThread().getName());
        //System.exit(0);

    }
    /** end-main */
}
